Diamond pattern in Python